package simple.array;

/**
 * <a href="https://leetcode.cn/problems/number-of-equivalent-domino-pairs/">1128. 等价多米诺骨牌对的数量</a>
 * 给你一个由一些多米诺骨牌组成的列表 dominoes。
 * 如果其中某一张多米诺骨牌可以通过旋转 0 度或 180 度得到另一张多米诺骨牌，我们就认为这两张牌是等价的。
 * 形式上，dominoes[i] = [a, b] 和 dominoes[j] = [c, d] 等价的前提是 a==c 且 b==d，或是 a==d 且 b==c。
 * 在 0 <= i < j < dominoes.length 的前提下，找出满足 dominoes[i] 和 dominoes[j] 等价的骨牌对 (i, j) 的数量。
 * 示例：
 *   输入：dominoes = [[1,2],[2,1],[3,4],[5,6]]
 *   输出：1
 * 提示：
 *   1 <= dominoes.length <= 40000
 *   1 <= dominoes[i][j] <= 9
 * @author 刘学松
 * @date 2023-07-24 9:40
 */
public class 等价多米诺骨牌对的数量 {
    public int numEquivDominoPairs(int[][] dominoes) {
        int total = 0;
        int[][] cnt = new int[10][10];
        for (int[] dominoe : dominoes) {
            ++cnt[dominoe[0]][dominoe[1]];
        }
        for (int i = 1; i < cnt.length; i++) {
            for (int j = i; j < cnt.length; j++) {
                int count = 0;
                if (i == j) {
                    count = cnt[i][j];
                } else {
                    count = cnt[i][j] + cnt[j][i];
                }
                if (count > 1) {
                    total += (count * (count - 1)) >> 1;
                }
            }
        }
        return total;
    }
}
